from selenium.webdriver.common.by import By

from web_objects.Calendar_web_objects.calendar_page import CalendarPage
from web_objects.base_page import BasePage


class MainPage(BasePage):
    _base_url = 'https://www.feishu.cn/'

    # 进入到日历页面
    def goto_calendar(self):
        self.find(By.CSS_SELECTOR, 'div._pp-product-container')
        self.find(By.XPATH, '//div[@title="日历"]')
        return CalendarPage(self.driver)
